itertools groupby multiple keys